Python code for Linear Regression